Node / ReticulumProjects / MeshChatX.git / files / tests / backend / test_database_snapshots.py

Displaying Raw • Download

Tff7b72import T7ee787os

Tff7b72import T7ee787shutil

Tff7b72import T7ee787tempfile

Tff7b72import T7ee787pytest

Tff7b72from T7ee787meshchatxT7ee787.T7ee787srcT7ee787.T7ee787backendT7ee787.T7ee787database Tff7b72import Te6edf3Database

Tf0883e@pytestTff7b72.Td2a8fffixture

Tff7b72def Td2a8fftemp_dirTb4b4b4(Tb4b4b4)Tb4b4b4:

Te6edf3dir_path Tff7b72= Te6edf3tempfileTff7b72.Td2a8ffmkdtempTb4b4b4(Tb4b4b4)

Tff7b72yield Te6edf3dir_path

Te6edf3shutilTff7b72.Td2a8ffrmtreeTb4b4b4(Te6edf3dir_pathTb4b4b4)

Tff7b72def Td2a8fftest_database_snapshot_creationTb4b4b4(Te6edf3temp_dirTb4b4b4)Tb4b4b4:

Te6edf3db_path Tff7b72= Te6edf3osTff7b72.Td2a8ffpathTff7b72.Td2a8ffjoinTb4b4b4(Te6edf3temp_dirTb4b4b4, Ta5d6ff"Ta5d6fftest.dbTa5d6ff"Tb4b4b4)

Te6edf3db Tff7b72= Te6edf3DatabaseTb4b4b4(Te6edf3db_pathTb4b4b4)

Te6edf3dbTff7b72.Td2a8ffinitializeTb4b4b4(Tb4b4b4)

T8b949e# Add some data

Te6edf3dbTff7b72.Td2a8ffexecute_sqlTb4b4b4(

Ta5d6ff"Ta5d6ffINSERT INTO config (key, value) VALUES (?, ?)Ta5d6ff"Tb4b4b4,

Tb4b4b4(Ta5d6ff"Ta5d6fftest_keyTa5d6ff"Tb4b4b4, Ta5d6ff"Ta5d6fftest_valueTa5d6ff"Tb4b4b4)Tb4b4b4,

Tb4b4b4)

T8b949e# Create snapshot

Te6edf3snapshot_name Tff7b72= Ta5d6ff"Ta5d6fftest_snapshotTa5d6ff"

Te6edf3dbTff7b72.Td2a8ffcreate_snapshotTb4b4b4(Te6edf3temp_dirTb4b4b4, Te6edf3snapshot_nameTb4b4b4)

Te6edf3snapshot_path Tff7b72= Te6edf3osTff7b72.Td2a8ffpathTff7b72.Td2a8ffjoinTb4b4b4(Te6edf3temp_dirTb4b4b4, Ta5d6ff"Ta5d6ffsnapshotsTa5d6ff"Tb4b4b4, Ta5d6fffTa5d6ff"Tffd700{Te6edf3snapshot_nameTffd700}Ta5d6ff.zipTa5d6ff"Tb4b4b4)

Tff7b72assert Te6edf3osTff7b72.Td2a8ffpathTff7b72.Td2a8ffexistsTb4b4b4(Te6edf3snapshot_pathTb4b4b4)

T8b949e# List snapshots

Te6edf3snapshots Tff7b72= Te6edf3dbTff7b72.Td2a8fflist_snapshotsTb4b4b4(Te6edf3temp_dirTb4b4b4)

Tff7b72assert Tffa657lenTb4b4b4(Te6edf3snapshotsTb4b4b4) Tff7b72== T79c0ff1

Tff7b72assert Te6edf3snapshotsTb4b4b4[T79c0ff0Tb4b4b4]Tb4b4b4[Ta5d6ff"Ta5d6ffnameTa5d6ff"Tb4b4b4] Tff7b72== Te6edf3snapshot_name

Tff7b72def Td2a8fftest_database_snapshot_restorationTb4b4b4(Te6edf3temp_dirTb4b4b4)Tb4b4b4:

Te6edf3db_path Tff7b72= Te6edf3osTff7b72.Td2a8ffpathTff7b72.Td2a8ffjoinTb4b4b4(Te6edf3temp_dirTb4b4b4, Ta5d6ff"Ta5d6fftest.dbTa5d6ff"Tb4b4b4)

Te6edf3db Tff7b72= Te6edf3DatabaseTb4b4b4(Te6edf3db_pathTb4b4b4)

Te6edf3dbTff7b72.Td2a8ffinitializeTb4b4b4(Tb4b4b4)

T8b949e# Add some data

Te6edf3dbTff7b72.Td2a8ffexecute_sqlTb4b4b4(Ta5d6ff"Ta5d6ffINSERT INTO config (key, value) VALUES (?, ?)Ta5d6ff"Tb4b4b4, Tb4b4b4(Ta5d6ff"Ta5d6ffv1Ta5d6ff"Tb4b4b4, Ta5d6ff"Ta5d6fforiginalTa5d6ff"Tb4b4b4)Tb4b4b4)

T8b949e# Create snapshot

Te6edf3dbTff7b72.Td2a8ffcreate_snapshotTb4b4b4(Te6edf3temp_dirTb4b4b4, Ta5d6ff"Ta5d6ffsnap1Ta5d6ff"Tb4b4b4)

Te6edf3snapshot_path Tff7b72= Te6edf3osTff7b72.Td2a8ffpathTff7b72.Td2a8ffjoinTb4b4b4(Te6edf3temp_dirTb4b4b4, Ta5d6ff"Ta5d6ffsnapshotsTa5d6ff"Tb4b4b4, Ta5d6ff"Ta5d6ffsnap1.zipTa5d6ff"Tb4b4b4)

T8b949e# Modify data

Te6edf3dbTff7b72.Td2a8ffexecute_sqlTb4b4b4(Ta5d6ff"Ta5d6ffUPDATE config SET value = ? WHERE key = ?Ta5d6ff"Tb4b4b4, Tb4b4b4(Ta5d6ff"Ta5d6ffmodifiedTa5d6ff"Tb4b4b4, Ta5d6ff"Ta5d6ffv1Ta5d6ff"Tb4b4b4)Tb4b4b4)

Te6edf3row Tff7b72= Te6edf3dbTff7b72.Td2a8ffproviderTff7b72.Td2a8fffetchoneTb4b4b4(Ta5d6ff"Ta5d6ffSELECT value FROM config WHERE key = ?Ta5d6ff"Tb4b4b4, Tb4b4b4(Ta5d6ff"Ta5d6ffv1Ta5d6ff"Tb4b4b4,Tb4b4b4)Tb4b4b4)

Tff7b72assert Te6edf3rowTb4b4b4[Ta5d6ff"Ta5d6ffvalueTa5d6ff"Tb4b4b4] Tff7b72== Ta5d6ff"Ta5d6ffmodifiedTa5d6ff"

T8b949e# Restore snapshot

Te6edf3dbTff7b72.Td2a8ffrestore_databaseTb4b4b4(Te6edf3snapshot_pathTb4b4b4)

T8b949e# Verify data is back to original

Te6edf3row Tff7b72= Te6edf3dbTff7b72.Td2a8ffproviderTff7b72.Td2a8fffetchoneTb4b4b4(Ta5d6ff"Ta5d6ffSELECT value FROM config WHERE key = ?Ta5d6ff"Tb4b4b4, Tb4b4b4(Ta5d6ff"Ta5d6ffv1Ta5d6ff"Tb4b4b4,Tb4b4b4)Tb4b4b4)

Tff7b72assert Te6edf3row Tff7b72is Tff7b72not Tff7b72None

Tff7b72assert Te6edf3rowTb4b4b4[Ta5d6ff"Ta5d6ffvalueTa5d6ff"Tb4b4b4] Tff7b72== Ta5d6ff"Ta5d6fforiginalTa5d6ff"

Tff7b72def Td2a8fftest_database_auto_backup_logicTb4b4b4(Te6edf3temp_dirTb4b4b4)Tb4b4b4:

T8b949e# This test verifies the loop logic if possible, or just the backup method

Te6edf3db_path Tff7b72= Te6edf3osTff7b72.Td2a8ffpathTff7b72.Td2a8ffjoinTb4b4b4(Te6edf3temp_dirTb4b4b4, Ta5d6ff"Ta5d6fftest.dbTa5d6ff"Tb4b4b4)

Te6edf3db Tff7b72= Te6edf3DatabaseTb4b4b4(Te6edf3db_pathTb4b4b4)

Te6edf3dbTff7b72.Td2a8ffinitializeTb4b4b4(Tb4b4b4)

T8b949e# Should create a timestamped backup

Te6edf3result Tff7b72= Te6edf3dbTff7b72.Td2a8ffbackup_databaseTb4b4b4(Te6edf3temp_dirTb4b4b4)

Tff7b72assert Ta5d6ff"Ta5d6ffdatabase-backupsTa5d6ff" Tff7b72in Te6edf3resultTb4b4b4[Ta5d6ff"Ta5d6ffpathTa5d6ff"Tb4b4b4]

Tff7b72assert Te6edf3osTff7b72.Td2a8ffpathTff7b72.Td2a8ffexistsTb4b4b4(Te6edf3resultTb4b4b4[Ta5d6ff"Ta5d6ffpathTa5d6ff"Tb4b4b4]Tb4b4b4)

Te6edf3backup_dir Tff7b72= Te6edf3osTff7b72.Td2a8ffpathTff7b72.Td2a8ffjoinTb4b4b4(Te6edf3temp_dirTb4b4b4, Ta5d6ff"Ta5d6ffdatabase-backupsTa5d6ff"Tb4b4b4)

Tff7b72assert Tffa657lenTb4b4b4(Te6edf3osTff7b72.Td2a8fflistdirTb4b4b4(Te6edf3backup_dirTb4b4b4)Tb4b4b4) Tff7b72== T79c0ff1